data_root=''
backbone=''   # "resnet18", "wide_resnet50"
size="256x256"
k=100